CUDA_VISIBLE_DEVICES=0 python3.5 train.py -p helmet -c 0 --batch_size 8 --lr 1e-5 --num_epochs 500 \
 --data_path /media/gzzn/Data/Datasets/ObjectDetection/helmet/
 --load_weights logs/helmet_coco/efficientdet-d0_9_15160.pth \
 --head_only True